import os

from experiment.DuoRecExp import DuoRecExp
from util.other_utils import load_yaml

if __name__ == '__main__':
    os.environ['CUDA_VISIBLE_DEVICES'] = '1'

    config_path = './config'

    config = load_yaml(os.path.join(config_path, 'config.yaml'))
    sensitive = load_yaml(os.path.join(config_path, 'sensitive.yaml'))

    database_config = sensitive['database_config']
    email_config = sensitive['email_config']

    exp = DuoRecExp(**config)
    exp.use_model = 'transform'
    exp.executive = 'xuhaoran'
    exp.data_load()
    exp.model_build()
    exp.train()
    exp.model_save()
    exp.evaluate()
    exp.dataset_update(database_config)
    exp.email(receiver='512139096@qq.com',
              email_config=email_config)
    exp.log_info()





